package leetcode._04_字符串;

import org.junit.Test;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * @author pppppp
 * @date 2022/4/12 10:23
 * 给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。
 * 异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。

 * 示例 1:
 * 输入: s = "cbaebabacd", p = "abc"
 * 输出: [0,6]
 * 解释:
 * 起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
 * 起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。
 *
 *  示例 2:
 * 输入: s = "abab", p = "ab"
 * 输出: [0,1,2]
 * 解释:
 * 起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。
 * 起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。
 * 起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。

 * 提示:
 * 1 <= s.length, p.length <= 3 * 104
 * s 和 p 仅包含小写字母

 */
public class _438_找到字符串中所有字母异位词 {

    @Test
    public void T_0() {
        String[][] nums = {{"cbaebabacd", "abc"}, {"abab", "ab"}};
        Integer[][] ans = {{0,6},{0,1,2}};
        for (int i = 0; i < nums.length; i++) {
            List<Integer> anagrams = findAnagrams(nums[i][0], nums[i][1]);
            for (int j = 0; j < anagrams.size(); j++) {
                System.out.println(ans[i][j].equals(anagrams.get(j)));
            }
        }
    }

    /*0.滑动窗口*/
    public List<Integer> findAnagrams(String s, String p) {
        ArrayList<Integer> res = new ArrayList<>();
        if (p.length() > s.length()) {
            return res;
        }
        HashMap<Character, Integer> tmap = new HashMap<>();
        HashMap<Character, Integer> curmap = new HashMap<>();
        char[] ch1 = p.toCharArray();
        int n = p.length(), count = 0;
        for (char c : ch1) {
            tmap.put(c, tmap.getOrDefault(c, 0) + 1);
        }
        char[] ch2 = s.toCharArray();
        for (int i = 0; i < n; i++) {
            if (tmap.containsKey(ch2[i])) {
                curmap.put(ch2[i], curmap.getOrDefault(ch2[i], 0) + 1);
                if (curmap.get(ch2[i]) <= tmap.get(ch2[i])) {
                    count++;
                }
            }
        }

        if (count == n) {
            res.add(0);
        }
        int l = 0, r = n;
        while (r < s.length()) {
            if(tmap.containsKey(ch2[r])){
                curmap.put(ch2[r], curmap.getOrDefault(ch2[r], 0) + 1);
                if (curmap.get(ch2[r]) <= tmap.get(ch2[r])) {
                    count++;
                }
            }
            if(tmap.containsKey(ch2[l])){
                curmap.put(ch2[l], curmap.get(ch2[l]) - 1);
                if (curmap.get(ch2[l]) < tmap.get(ch2[l])) {
                    count--;
                }
            }
            l++;
            r++;
            if (count == n) {
                res.add(l);
            }
        }
        return res;
    }
}
